# -*- coding: utf-8 -*-

import warnings

import pandas as pd
from sklearn import feature_selection

from ..base import CommonFunction


class FeatureSelection(CommonFunction):
    def variance_threshold(self, threshold=0):
        selector = feature_selection.VarianceThreshold(threshold=threshold)
        self._data = pd.DataFrame(selector.fit_transform(self._data))
        return selector

    def correlation_coefficient(self):
        pass

    def chi2(self):
        pass

    def mutual_information(self):
        pass

    def recursive_elimination(self):
        pass

    def select_from_penalty(self):
        pass

    def select_from_tree_model(self):
        pass
